package ai.zixing.sparsearray;

/**
 * 稀疏数组
 */
public class SparesArray {
    public static void main(String[] args) {
        // 创建一个原始二维数组 11 * 11
        // 0 ： 表示没有子， 1 表示黑子，2表示白子
        int chessArr1[][] = new int[11][11];
        chessArr1[1][2] = 1;
        chessArr1[2][3] = 2;
        // 输出原始的二维数组
        System.out.println("原始的二维数组");
        for (int[] row : chessArr1) {
            for (int data : row) {
                System.out.printf("%d\t", data);
            }
            System.out.println();
        }
        // 将二维数组转为 稀疏数组
        // 1. 遍历二维数据，得到非0数据的个数
        int sum = 0;
        for (int i = 0; i < 11; i++) {
            for (int j = 0; j < 11; j++) {
                if (chessArr1[i][j] != 0) {
                    sum++;
                }
            }
        }
        System.out.println("sum = " + sum);
        // 2.创建对应的稀疏数组
        int sparesArr[][] = new int[sum + 1][3];
        // 给稀疏数组赋值
        sparesArr[0][0] = 11;
        sparesArr[0][1] = 11;
        sparesArr[0][2] = sum;
        // 遍历二维数组，将非零值赋值给稀疏数组
        // count 记录是第几个非0数据
        int count = 0;
        for (int i = 0; i < 11; i++) {
            for (int j = 0; j < 11; j++) {
                if (chessArr1[i][j] != 0) {
                    count++;
                    sparesArr[count][0] = i;
                    sparesArr[count][1] = j;
                    sparesArr[count][2] = chessArr1[i][j];
                }
            }
        }
        System.out.println();
        System.out.println("输出稀疏数组为 ： ");
        for (int i = 0; i < sparesArr.length; i++) {
            System.out.printf("%d\t%d\t%d\t\n", sparesArr[i][0], sparesArr[i][1], sparesArr[i][2]);
        }
        System.out.println();

        // 将稀疏数组恢复为二维数据
        System.out.println("将稀疏数组恢复为二维数据");
        // 1. 读取第一行，创建二维数组
        int chessArr2[][] = new int[sparesArr[0][0]][sparesArr[0][1]];
        // 2.读取稀疏数组后几行数据（从第二行开始），并赋值给二维数组
        for (int i = 1; i < sparesArr.length; i++) {
            chessArr2[sparesArr[i][0]][sparesArr[i][1]] = sparesArr[i][2];
        }

        // 恢复后的二维数组
        System.out.println("恢复后的二维数组");
        for (int[] row : chessArr2) {
            for (int data : row) {
                System.out.printf("%d\t", data);
            }
            System.out.println();
        }
    }
}
